ram = nn.RecurrentAttention(rnn, action, nStep, hiddenSize)